#!/usr/bin/env python
# encoding: utf-8
"""
@author:zwb
@file: logger.py
@contact: zwb15083976291@163.com
@time: 2021/10/09
"""

import os
from datetime import datetime

BASE_DIR = os.environ.get("BASE_DIR", "")
if not BASE_DIR:
    assert BASE_DIR, 'BASE_DIR未配置！'

env = os.environ.get('BLOG_PROJECT_ENV', 'dev')
DATE = str(datetime.now().date())

LOG_DIR = os.path.join(BASE_DIR, 'logs')
if not os.path.exists(LOG_DIR):
    os.mkdir(LOG_DIR)

LOGGING = {
    'version': 1,
    'disable_existing_loggers': False,
    'formatters': {
        'standard': {
            'format': '[%(levelname)s][%(asctime)s][%(threadName)s:%(thread)d][task_id:%(name)s][%(filename)s:%(lineno)d]'
                      '[%(message)s]'
        },
        'simple': {
            'format': '[%(levelname)s][%(asctime)s][%(filename)s:%(lineno)d]%(message)s'
        },
        'collect': {
            'format': '%(message)s'
        },
        'unknown': {
            'format': '[%(levelname)s][%(asctime)s][%(threadName)s:%(thread)d][task_id:%(name)s][%(message)s]'
        },
    },
    'filters': {
        'require_debug_true': {
            '()': 'django.utils.log.RequireDebugTrue',
        },
    },
    'handlers': {
        'console': {
            'level': 'ERROR',
            # 'filters': ['require_debug_true'],  # 只有在Django debug为True时才在屏幕打印日志
            'class': 'logging.StreamHandler',
            'formatter': 'standard'
        },
        'info': {
            'level': 'INFO',
            'class': 'logging.handlers.RotatingFileHandler',  # 保存到文件，根据文件大小自动切
            'filename': os.path.join(LOG_DIR, f"{DATE}_info.log"),  # 日志文件
            'maxBytes': 1024 * 1024 * 50,  # 日志大小 50M
            'backupCount': 3,  # 备份数为3  xx.log --> xx.log.1 --> xx.log.2 --> xx.log.3
            # 'when': 'D',  # 每天一切， 可选值有S/秒 M/分 H/小时 D/天 W0-W6/周(0=周一) midnight/如果没指定时间就默认在午夜
            'formatter': 'standard',
            'encoding': 'utf-8',
        },
        'error': {
            'level': 'ERROR',
            'class': 'logging.handlers.RotatingFileHandler',  # 保存到文件，自动切
            'filename': os.path.join(LOG_DIR, f"{DATE}_error.log"),  # 日志文件
            'maxBytes': 1024 * 1024 * 5,  # 日志大小 50M
            'backupCount': 5,
            'formatter': 'standard',
            'encoding': 'utf-8',
        },
        'unknown_error': {
            'level': 'ERROR',
            'class': 'logging.handlers.RotatingFileHandler',  # 保存到文件，自动切
            'filename': os.path.join(LOG_DIR, f"{DATE}_unknown_error.log"),  # 日志文件
            'maxBytes': 1024 * 1024 * 5,  # 日志大小 50M
            'backupCount': 5,
            'formatter': 'unknown',
            'encoding': 'utf-8',
        },
        'warning': {
            'level': 'WARNING',
            'class': 'logging.handlers.RotatingFileHandler',  # 保存到文件，自动切
            'filename': os.path.join(LOG_DIR, f"{DATE}_warning.log"),  # 日志文件
            'maxBytes': 1024 * 1024 * 5,  # 日志大小 50M
            'backupCount': 5,
            'formatter': 'standard',
            'encoding': 'utf-8',
        },
    },
    'loggers': {
        'default': {  # 名为 'collect'的logger还单独处理
            'handlers': ['error', 'warning', 'info'],
            'level': 'INFO',
            'propagate': False
        },
        'default.info': {  # 名为 'collect'的logger还单独处理
            'handlers': ['info'],
            'level': 'INFO',
            'propagate': False
        },
        'default.warning': {  # 名为 'collect'的logger还单独处理
            'handlers': ['warning'],
            'level': 'WARNING',
            'propagate': False
        },
        'default.error': {  # 名为 'collect'的logger还单独处理
            'handlers': ['error'],
            'level': 'ERROR',
            'propagate': False
        },
        'unknown_error': {  # 名为 'unknown_error'的logger还单独处理
            'handlers': ['unknown_error'],
            'level': 'ERROR',
            'propagate': False
        },
    },
}
